/*
 *    This file is part of SSHTunneler.
 *
 *    Copyright 2009 Jean-Sebastien Gelinas <calestar@gmail.com>
 *
 *    SSHTunneler is free software: you can redistribute it and/or modify
 *    it under the terms of the GNU General Public License as published by
 *    the Free Software Foundation, either version 3 of the License, or
 *    (at your option) any later version.
 *
 *    SSHTunneler is distributed in the hope that it will be useful,
 *    but WITHOUT ANY WARRANTY; without even the implied warranty of
 *    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 *    GNU General Public License for more details.
 *
 *    You should have received a copy of the GNU General Public License
 *    along with SSHTunneler.  If not, see <http://www.gnu.org/licenses/>.
 *
 */
package sshtunneler.server_tunnel.tree;

import javax.swing.JPopupMenu;

import sshtunneler.library.Server;
import sshtunneler.library.ServerContainer;

public class ServerTunnelRootNode extends MyServerTunnelTreeNode {
	private static final long serialVersionUID = -9175805529722030731L;
	private GenericServerTunnelTree tree;
	private ServerContainer<? extends Server> container;

	public ServerTunnelRootNode(GenericServerTunnelTree tree,
			ServerContainer<? extends Server> container) {
		this.container = container;
		this.tree = tree;
		this.setAllowsChildren(true);
		this.setUserObject("Servers");
	}

	public ServerNode addServerNode(Server server) {
		ServerNode newNode = new ServerNode(server, this.tree);

		this.add(newNode);

		return newNode;
	}

	@Override
	public void refresh() {
		this.removeAllChildren();
		for (Server server : this.container.getServers()) {
			this.addServerNode(server).refresh();
		}
	}

	@Override
	public JPopupMenu getPopup() {
		return this.tree.get_root_menu(this);
	}
}
